




source  /usr/local/miniconda3/etc/profile.d/conda.sh 



python -c "import torch;print(torch.cuda.is_available(), torch.__version__,torch.cuda.device_count())"

conda activate atlasnet
#* train 

bs=256;
# CUDA_VISIBLE_DEVICES='1' python train.py \
# --shapenet13 --dir_name logs/atlasnet_autoencoder_25_squares \
# --nb_primitives 25 --template_type "SQUARE" \
# --batch_size $bs  --batch_size_test $bs 




 
 #* train 
CUDA_VISIBLE_DEVICES='1' python train.py \
--shapenet13 --dir_name logs/atlasnet_autoencoder_25_squares \
--nb_primitives 25 --template_type "SQUARE" \
--batch_size $bs  --batch_size_test $bs --run_single_eval --no_metro \
--reload_model_path logs/atlasnet_autoencoder_25_squares/network.pth



  



#* error with this parameter : " > logs/train.log"